超坑！安装CUDA与pytorch版本匹配的问题

慢悠悠地装了很久，觉得CUDA和pytorch官网标注的当前该pytorch所匹配的CUDA应该是完全一致的，所以一开始就安装 10.1.105版本的CUDA，pytorch为1.4（官方说适配CUDA版本为10.1嘛~）：

但是事实证明，不行。 最后还是降级安装pytorch，从1.4（对应CUDA10.1）到1.2版本的pytorch（对应CUDA10.0），还是不行，最后：

显卡驱动版本号：431.56（安装之前一定要看自己的显卡驱动版本号，然后对应下载CUDA与cudnn进行安装~） CUDA版本：10.1.105 pytorch版本：1.1.0 （对应CUDA 10.0）

注意：本身的CUDA版本与我们在虚拟环境中安装的cudatoolkit包是没有太大关系的，一般安装pytorch时需要考虑的cuda版本指的应该是虚拟环境中安装的cudatoolkit包的版本（个人经验激情发言，不负责任噢）。

测试成功！！ 所以如果安装好了CUDA之后，可以选择pytorch官方说明的适配CUDA版本低一些的来进行安装~ 目前俺正在使用的两种版本是： ① cudatoolkit：10.1.243 pytorch1.6.0（pytorch1.3.1也可以） cudnn是7.6.4 ② cudatoolkit10.0.130 pytorch:1.1.0

附：安装tensorflow-gpu版本：